var test = function (server) {
  // 导入WebSocket模块:
  const WebSocket = require("ws");

  // 引用Server类:
  const WebSocketServer = WebSocket.Server;

  // 实例化:
  const wss = new WebSocketServer({
    server,
  });

  const clients = [];

  wss.on("connection", function (ws) {
    clients.push(ws);
    ws.wss = wss;
    ws.on("message", function (msg) {
      msg = msg.toString();
      msg = JSON.parse(msg);
      console.log(5555,msg);
      if (msg.type == "login") {
        sendAll(msg.user + "加入聊天室");
      }else if(msg.msg){
        sendAll(msg.name+'：'+msg.msg)
      }
    });
  });
  wss.on("close", function (ws) {
    console.log("有人走了");
  });

  function sendAll(msg) {
    for (var i = 0; i < clients.length; i++) {
      clients[i].send(msg);
    }
  }
};

module.exports = test;

